import os


PNET_MODEL_STORE_DIR = "/media/handsome/data/hanson/WIDER_train/model_store/heils_wider/pnet"
RNET_MODEL_STORE_DIR = "/media/handsome/data/hanson/WIDER_train/model_store/heils_wider/rnet"
ONET_MODEL_STORE_DIR = "/media/handsome/data/hanson/WIDER_train/model_store/heils_wider/onet"

ANNO_STORE_DIR = "/media/handsome/data/hanson/WIDER_train/anno_store/heils_wider"


LOG_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))+"/log"


USE_CUDA = True


TRAIN_BATCH_SIZE = 512
FREQUENCE = 200
TRAIN_LR = 0.002

END_EPOCH = 500
NUM_WORKERS = 16

PNET_POSTIVE_ANNO_FILENAME = "pos_12.txt"
PNET_NEGATIVE_ANNO_FILENAME = "neg_12.txt"
PNET_PART_ANNO_FILENAME = "part_12.txt"
PNET_LANDMARK_ANNO_FILENAME = "landmark_12.txt"


RNET_POSTIVE_ANNO_FILENAME = "pos_24.txt"
RNET_NEGATIVE_ANNO_FILENAME = "neg_24.txt"
RNET_PART_ANNO_FILENAME = "part_24.txt"
RNET_LANDMARK_ANNO_FILENAME = "landmark_24.txt"


ONET_POSTIVE_ANNO_FILENAME = "pos_48.txt"
ONET_NEGATIVE_ANNO_FILENAME = "neg_48.txt"
ONET_PART_ANNO_FILENAME = "part_48.txt"
ONET_CELEBA_LANDMARK_ANNO_FILENAME = "celebA_landmark_48.txt"
ONET_BACKLIGHT_LANDMARK_ANNO_FILENAME = "backlight_landmark_48.txt"


PNET_TRAIN_IMGLIST_FILENAME = "imglist_anno_12.txt"
RNET_TRAIN_IMGLIST_FILENAME = "imglist_anno_24.txt"
# RNET_TRAIN_IMGLIST_FILENAME = "pos_24.txt"
ONET_TRAIN_IMGLIST_FILENAME = "imglist_anno_48.txt"
# ONET_TRAIN_IMGLIST_FILENAME = "landmark_48.txt"